from 辅助工具 import *

#主要流程
def main():
    name = input("请输入用户名：")
    password = int(input("请输入密码："))
    a = login(name, password)  # a接收登录的返回值，如果是1，普通用户登陆成功，2管理员用户登陆成功，-1登陆失败

    if a == 1: #普通用户
        while True:
            print("1.查询所有 2.按条件查询 3.推出")
            c = int(input())
            if c == 1:
                selectAll()
            if c == 2:
                print("0.根据ID查询 1.根据支出类别查询 2.根据支付方式查询 3.根据消费金额查询 4.根据消费时间查询 5.根据备注查询")
                d = int(input("选择查询方式："))
                e = input("输入查询的内容：")
                selectBy(d,e)
            if c == 3:
                break

    if a == 2: #管理员用户
        while True:
            print("1.添加账务 2.编辑账务 3.删除账务 4.查询账务 5 退出系统")
            b = int(input("请输入要操作的功能的序号："))

            #1.添加账务
            if b == 1:
                type = input("请输入支出类别：")
                method = input("请输入支付方式：")
                money = float(input("请输入消费金额："))
                date = input("请输入消费时间：")
                comment = input("请输入备注：")
                addAccount(type,method,money,date,comment)

            #2.编辑账务
            if b == 2:
                id = int(input('请输入要编辑账务的id：'))
                type = input("请输入支出类别：")
                method = input("请输入支付方式：")
                money = float(input("请输入消费金额："))
                date = input("请输入消费时间：")
                comment = input("请输入备注：")
                editAccount(id,type,method,money,date,comment)

            #3.删除账务
            if b == 3:
                f = int(input("请输入要删除的ID"))
                delAccount(f)

            #4.查询账务
            if b == 4:
                selectAll()
            if b == 5:
                break
main()